from data.paris import loadparis
from data.oxford import loadoxford
from data.newcollege import loadnewcollege
from data.kitti import loadkitti
from data.bdd100k import loadbdd100k
from data.aachen import loadaachen
from data.cityscape import loadcityscape
from data.mapillary import loadmapillary

dataloader = {"aachen": loadaachen,
              "bdd100k": loadbdd100k,
              "cityscape": loadcityscape,
              "kitti": loadkitti,
              "newcollege": loadnewcollege,
              "oxford": loadoxford,
              "paris": loadparis,
              "mapillary": loadmapillary}

def LoadData(data, data_path, mode='train'):
    assert data in dataloader.keys()
    return dataloader[data](data_path, mode)
